from celery_sqlalchemy_scheduler.schedulers import DatabaseScheduler


class CustomScheduler(DatabaseScheduler):

    def apply_entry(self, entry, producer=None):
        entry.kwargs.update({'__entry_name': entry.name})
        super(CustomScheduler, self).apply_entry(entry, producer)